Improving on stuff

Signed-off-by: Kenneth Lim <kenlimmj@gmail.com>